// Copyright 2019 by 马万里. All rights reserved.
// 开发团队 ：   鸡中之霸
// 开发人员 ：   马万里
// 开发时间 ：   2019-02-19 15:53
// 文件名称 ：   user.go
// 工程名称 ：   beego_project
// 开发工具 ：   GoLand
//

package models

import (
	"github.com/jinzhu/gorm"
)

type User struct {
	gorm.Model
	Name   string `gorm:"unique_index"`
	Email  string `gorm:"unique_index"`
	Pwd    string
	Avatar string
	Role   int `gorm:"default:0"` // 1代表管理员，0代表普通用户，gorm的坑点，如果设置为0，那么就回去取默认值。
}

func QueryUserByEmailAndPwd(email, pwd string) (user User, err error) {
	return user, db.Where("email = ? and pwd = ?", email, pwd).Take(&user).Error
}

func QueryUserByName(name string) (user User, err error) {
	return user, db.Where("name = ?", name).Take(&user).Error
}

func QueryUserByEmail(email string) (user User, err error) {
	return user, db.Where("email = ?", email).Take(&user).Error
}

func SaveUser(user *User) error {
	return db.Save(user).Error
}
